前言:
最近需要用到訓練模型來完成一些圖片辨識,以 TensorFlow 來做為這次學習的重點,並參照了兩個方式來實行這次的作業
正文:
從終端機安裝 TensorFlow
# CPU
pip install tensorflow
# GPU
pip install tensorflow-gpu
pip install tensorflow_datasets
或是用Anaconda來安裝
# CPU
conda install tensorflow
# GPU
conda install tensorflow-gpu
conda install tensorflow_datasets
程式碼:
方案1
import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf
import tensorflow_datasets as tfds
from tensorflow.keras import layers
方案2
from tensorflow.python.keras import backend as K
from tensorflow.python.keras.models import Model
from tensorflow.python.keras.layers import Flatten, Dense, Dropout
from tensorflow.python.keras.applications.resnet import ResNet50
from tensorflow.python.keras.optimizers import *
from tensorflow.python.keras.preprocessing.image import ImageDataGenerator
numpy: 專門處理矩陣、向量的套件。
matplotlib.pyplot: 繪圖套件。
tensorflow: 機器學習套件,可訓練模型、存取模型。
tensorflow_datasets: 用以下載公開資料集 mnist。
tensorflow.keras: 專注於神經網路的機器學習套件,使用上比 tensorflow 更簡單直觀。